package cn.tedu.lucky.draw.server.job.config;

import cn.tedu.lucky.draw.server.job.listener.ActivityListener;
import cn.tedu.lucky.draw.server.job.processor.ActivityItemProcessor;
import cn.tedu.lucky.draw.server.protrocol.po.Activity;
import cn.tedu.lucky.draw.server.repository.ActivityRepository;
import org.springframework.batch.core.Job;
import org.springframework.batch.core.Step;
import org.springframework.batch.core.configuration.annotation.JobBuilderFactory;
import org.springframework.batch.core.configuration.annotation.StepBuilderFactory;
import org.springframework.batch.core.launch.support.RunIdIncrementer;
import org.springframework.batch.item.ItemWriter;
import org.springframework.batch.item.data.RepositoryItemWriter;
import org.springframework.batch.item.file.FlatFileItemReader;
import org.springframework.batch.item.file.builder.FlatFileItemReaderBuilder;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.io.PathResource;

import java.io.File;

/**
 * 定义作业：异步读取csv文件到数据库
 */

@Configuration
public class CsvToDBJobConfig {

    @Autowired
    private JobBuilderFactory jobBuilderFactory;

    @Autowired
    private StepBuilderFactory stepBuilderFactory;

    @Autowired
    private ActivityRepository activityRepository;

    //itemReader
    @Bean
    public FlatFileItemReader<Activity> itemReader() {
        return new FlatFileItemReaderBuilder<Activity>()
                .name("activityIteamReader")//为这个iteamReader实例设置一个名称
                .saveState(false)//不保存状态，如果作业失败重启，它不会从上次失败的地方继续执行
                .resource(new PathResource(new File("doc/activity.csv").getAbsolutePath()))//指定读取的文件资源
                .delimited()//指定这个文件是用分隔符分隔的文本文件
                .names("name", "description", "img", "status", "sort")//定义csv文件中每列数据对应的字段名
                .targetType(Activity.class)//设置目标类型为Activity对象，每读取一行数据都会被转换为一个Activity实例
                .linesToSkip(1)//设置在开始读取数据之前要跳过的行数，这里是设置为1，意味着跳过文件的第一行，第一行作为标题行
        .build();
    }

    @Bean
    public ActivityItemProcessor activityItemProcessor(){
        return new ActivityItemProcessor();
    }

    //itemWriter
    @Bean
    public ItemWriter<Activity> activityItemWriter(){
        //RepositoryItemWriter专门配合Spring Data JPA进行使用的，能够将批处理中的数据批量写入到数据库
        RepositoryItemWriter<Activity> itemWriter = new RepositoryItemWriter<>();
        //将activityRepository实例注入到writer中，通过这个种方式，能够利用该持久层将数据持久化到数据库
        itemWriter.setRepository(activityRepository);
        //每当写入数据时，都会调用activityRepository的save方法
        itemWriter.setMethodName("save");
        //返回配置好的实例对象
        return itemWriter;
    }

    @Bean
    public Step csvToDBStep(){
        return stepBuilderFactory.get("csvToDBStep")
                /**
                 * 该方法定义了每个“块”大小，每次处理的数据项的数量
                 * 当前表示每次处理10000个Activity对象
                 */
                .<Activity,Activity>chunk(10000)
                .reader(itemReader())
                .processor(activityItemProcessor())
                .writer(activityItemWriter())
                .build();
    }

    @Bean
    public ActivityListener activityListener(){
        return new ActivityListener();
    }

    @Bean
    public Job csvToDBJob(){
        return jobBuilderFactory.get("csvToDBJob")
                .start(csvToDBStep())
                /**
                 * 每次作业运行时都会自动增加一个运行ID
                 * 以此来确保可以重复运行作业
                 */
                .incrementer(new RunIdIncrementer())
                .listener(activityListener())
                .build();
    }

}
